# 从Excel表格相应表单的指定列抽取出所有内容按指定格式保存到TXT文档中

import xlrd as xr
import xlwt as xw
import argparse

parser = argparse.ArgumentParser()
parser.add_argument('source', type=str, help="source: 操作表格路径")
parser.add_argument('dst', type=argparse.FileType('a'), help="dst: 保存文件路径")
parser.add_argument('-sheet', type=int, default=0, help="sheet: 操作表单的索引号，默认为0")
parser.add_argument('-column', type=int, default=0, help="column: 操作表单中对应的列号，默认为0")
parser.add_argument('-srow', type=int, default=0, help="srow: 起始行号，默认为0")
parser.add_argument('-erow', type=int, help="erow: 结束行号，默认为最大行号")
args = parser.parse_args()

book = xr.open_workbook(args.source)    # 打开指定的表格文件， 如：17软件计应分组计划.xlsx
sheet = book.sheet_by_index(args.sheet) # 获取指定序号的表单，如：17安卓.txt
row_end = args.erow if args.erow else sheet.nrows

for i in range(args.srow, row_end):
    if i < row_end-1:
        args.dst.write(str(int(sheet.cell(i, args.column).value))+',')
    else:
        args.dst.write(str(int(sheet.cell(i, args.column).value)))

args.dst.close()
